from selenium import webdriver
from selenium.webdriver.common.by import By
import time

# 初始化浏览器驱动
driver = webdriver.Chrome()

# 设置隐式等待时间为10秒
driver.implicitly_wait(10)

# 访问一个网址
driver.get("https://www.jd.com")

try:
    # 查找一个元素（例如搜索框）
    search_input = driver.find_element(By.ID, "key")
    print("目标元素已找到")

    # 截图保存
    driver.save_screenshot("wait_page04.png")
    print("截图已保存")

except Exception as e:
    print(f"找不到元素：{e}")

finally:
    # 关闭浏览器
    driver.quit()